import os
import subprocess

def activate_conda_env():
    """Activate the suanfamama conda environment"""
    # Get the conda activation command for the current OS
    if os.name == 'nt':  # Windows
        activate_cmd = "conda activate suanfamama"
    else:  # Unix/Linux/MacOS
        activate_cmd = "source activate suanfamama"
    
    try:
        subprocess.run(activate_cmd, shell=True, check=True)
        print("✅ Activated suanfamama environment")
    except subprocess.CalledProcessError:
        print("❌ Failed to activate environment")
        print("Please run: conda activate suanfamama")

if __name__ == "__main__":
    activate_conda_env() 